linear regression with keras
# Tag:
Linear Regression
import numpy as np import matplotlib.pyplot as plt # 텐서플로의 케라스 API에서 필요한 함수들을 불러옵니다. from tensorflow.keras.models import Sequential from tensorflow.keras.layers import Dense x = np.array([2, 4, 6, 8]) y = np.array([81, 93, 91, 97]) model = Sequential() # 출력 값, 입력 변수, 분석 방법에 맞게끔 모델을 설정합니다. # multiple linear regession이라면, input_dim=2,3,4... 등이 된다. model.add(Dense(1, input_dim=1, activation='linear')) # 오차 수정을 위해 경사 하강법(sgd)을, 오차의 정도를 판단하기 위해 # 평균 제곱 오차(mse)를 사용합니다. model.compile(optimizer='sgd', loss='mse') # 오차를 최소화하는 과정을 2000번 반복합니다. model.fit(x, y, epochs=2000) plt.scatter(x, y) plt.plot(x, model.predict(x), 'r') # 예측 결과를 그래프로 나타냅니다. plt.show() # 임의의 시간을 집어넣어 점수를 예측하는 모델을 테스트 hour = 7 prediction = model.predict([hour]) print("%.f시간을 공부할 경우의 예상 점수는 %.02f점입니다." % (hour, prediction))>)